'''
Assignment: Create PT Practice #4 (from Display3)
Programmers:
'''


from codex import *
from time import sleep


messages = ["Press Up", "Press Down", "Press Left", "Press Right",
            "Press A", "Press B", "Press Left", "Press Down"]
btns = [BTN_U, BTN_D, BTN_L, BTN_R, 
        BTN_A, BTN_B, BTN_L, BTN_D]


delay = 1
count = 0


# One function for game play
def play_game():
    global count
    for ind in range(len(messages)):
        message = messages[ind]
        button = btns[ind]
        display.show(message)
        sleep(delay)


        pressed = buttons.is_pressed(button)
        if pressed:
            pixels.set(ind%4, GREEN)
            count = count + 1
        else:
            pixels.set(ind%4, RED)
        sleep(delay)
        pixels.set(ind%4, BLACK)


def ending(count):
    # turn off all pixels and clear screen
    for pix in range(4):
        pixels.set(pix, BLACK)
    display.clear()


    if count == len(btns):
        end_message = "You won!"
        col = GREEN
    elif count == 0:
        end_message = "You lost"
        col = RED
    else:
        end_message = "Keep trying"
        col = BLUE
    
    display.draw_text(end_message, scale=3, x=30, y=100, color=col)
    # running pixel lights
    for num in range(30):
        pixels.set(num%4, col)
        sleep(0.2)
        pixels.set(num%4, BLACK)


# Main Program
play_game()
ending(count)